import os

def generate_markdown_file():
    # 获取output_frames目录中的文件列表
    frames_dir = 'output_frames'
    files = os.listdir(frames_dir)
    
    # 过滤出.jpg文件并按文件名排序
    image_files = sorted([f for f in files if f.endswith('.jpg')])
    
    # 创建并写入frames.md文件
    with open(os.path.join(frames_dir, 'frames.md'), 'w') as f:
        for i, image_file in enumerate(image_files, start=1):
            f.write(f'![{image_file}]({image_file})\n')
            if i < len(image_files):  # 在每两张图片之间添加序号
                f.write(f'### Fig. {i} {image_file}\n\n')

if __name__ == '__main__':
    generate_markdown_file()